package thread;


import java.util.concurrent.PriorityBlockingQueue;

// 代表一个定时器中的任务
class MyTask implements Comparable<MyTask> {
    // 要执行的任务的内容
    private Runnable runnable;
    // 表示任务要在啥时候执行 (使用毫秒时间戳来表示)
    private long time;

    public MyTask(Runnable runnable, long time) {
        this.runnable = runnable;
        this.time = time;
    }

    public void run() {
        runnable.run();
    }

    @Override
    public int compareTo(MyTask o) {
        // 返回 <0 >0 或者 0
        // this 比 o 小, 返回 <0
        // ...
        return (int) (this.time - o.time);
    }

    public long getTime() {
        return this.time;
    }
}

// 自己实现的定时器
class MyTimer {
    // 扫描线程
    private Thread t = null;

    // 有一个阻塞优先级队列, 来保存任务
    private PriorityBlockingQueue<MyTask> queue = new PriorityBlockingQueue<>();

    public MyTimer() {
        t = new Thread(() -> {
            while (true) {
                try {
                    synchronized (this) {
                        // 取出队首元素, 检查看看队首元素的任务是否到时间了
                        // 时间没到, 就把任务塞回去
                        // 时间到了, 就把任务进行执行
                        MyTask myTask = queue.take();
                        long curTime = System.currentTimeMillis();
                        if (curTime < myTask.getTime()) {
                            // 还没到点
                            queue.put(myTask);

                            this.wait(myTask.getTime() - curTime);

                        } else {
                            // 时间到了, 执行任务
                            myTask.run();
                        }
                    }
                } catch (InterruptedException e) {
                    throw new RuntimeException(e);
                }
            }
        });
        t.start();
    }

    /**
     * @param runnable 任务内容
     * @param after    任务在多少毫秒之后执行
     */
    public void schedule(Runnable runnable, long after) {
        // 这里时间上的换算, 要加上当前时间
        MyTask task = new MyTask(runnable, System.currentTimeMillis() + after);
        queue.put(task);
        synchronized (this) {
            this.notify();
        }
    }
}

public class ThreadDemo25 {
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("任务1");
            }
        }, 1000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("任务2");
            }
        }, 2000);
    }
}
